package com.tanhua.dubbo.server.api;

import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.convert.Convert;
import cn.hutool.core.date.DateUtil;
import cn.hutool.core.util.ObjectUtil;
import com.alibaba.dubbo.config.annotation.Service;
import com.tanhua.dubbo.server.pojo.RecommendUser;
import com.tanhua.dubbo.server.pojo.Visitors;
import com.tanhua.dubbo.server.vo.PageInfo;
import org.bson.types.ObjectId;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.domain.Sort;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.data.redis.core.RedisTemplate;

import java.util.List;

@Service(version = "1.0.0")
public class VisitorsApiImpl implements VisitorsApi {

    @Autowired
    private MongoTemplate mongoTemplate;

    @Autowired
    private RedisTemplate redisTemplate;

    private static final String VISITOR_REDIS_KEY = "VISITOR_USER";

    /**
     * 保存访客数据
     *
     * @param userId        我的id
     * @param visitorUserId 访客id
     * @param from          来源
     * @return
     */
    @Override
    public String saveVisitor(Long userId, Long visitorUserId, String from) {
        // 校验, 如果都为空, 就直接返回null
        if (!ObjectUtil.isAllNotEmpty(userId, visitorUserId, from)) {
            return null;
        }

        // 查询访客用户在今天是否已经记录过, 如果已经记录过, 就不再记录
        String today = DateUtil.today();
        Long minDate = DateUtil.parseDateTime(today + " 00:00:00").getTime();
        Long maxDate = DateUtil.parseDateTime(today + " 23:59:59").getTime();
        // 条件: 传递过来的我的id和来访用户id都相等, 时间符合要求
        Query query = Query.query(Criteria.where("userId").is(userId)
                .and("visitorUserId").is(visitorUserId)
                // 时间大于等于最小时间, 小于等于最大时间
                .andOperator(Criteria.where("date").gte(minDate), Criteria.where("date").lte(maxDate)));
        long count = mongoTemplate.count(query, Visitors.class);
        if (count > 0) {
            //今天已经记录过
            return null;
        }
        // 封装数据
        Visitors visitors = new Visitors();
        visitors.setId(ObjectId.get());
        visitors.setUserId(userId);
        visitors.setVisitorUserId(visitorUserId);
        visitors.setFrom(from);
        visitors.setDate(System.currentTimeMillis());

        //存储数据
        mongoTemplate.save(visitors);

        return visitors.getId().toHexString();
    }

    /**
     * 查询我的访客数据，存在2种情况：
     * 1. 我没有看过我的访客数据，返回前5个访客信息
     * 2. 之前看过我的访客，从上一次查看的时间点往后查询5个访客数据
     *
     * @param userId 我的id
     * @return
     */
    @Override
    public List<Visitors> queryMyVisitor(Long userId) {
        // 查询前5个访客数据，按照访问时间倒序排序
        // 如果用户已经查询过列表，记录查询时间，后续查询需要按照这个时间往后查询

        // 上一次查询列表的时间
        Long date = Convert.toLong(redisTemplate.opsForHash().get(VISITOR_REDIS_KEY, String.valueOf(userId)));

        // 查询前5个访客数据，按照访问时间倒序排序
        PageRequest pageRequest = PageRequest.of(0,5, Sort.by(Sort.Order.desc("date")));
        Query query = Query.query(Criteria.where("userId").is(userId)).with(pageRequest);
        // 如果时间不为空, 则表示用户已经查过了, 后续查询需要按照这个时间往后查询
        if (ObjectUtil.isNotEmpty(date)){
            query.addCriteria(Criteria.where("date").gte(date));
        }
        List<Visitors> visitorsList = mongoTemplate.find(query, Visitors.class);

        //查询每个来访用户的缘分值
        for (Visitors visitors : visitorsList) {
            Query queryScore = Query.query(Criteria.where("userId").is(userId).and("toUserId").is(visitors.getVisitorUserId()));
            RecommendUser recommendUser = mongoTemplate.findOne(queryScore, RecommendUser.class);
            if (ObjectUtil.isNotEmpty(recommendUser)){
                visitors.setScore(recommendUser.getScore());
            }else {
                //默认得分
                visitors.setScore(77d);
            }
        }
        return visitorsList;
    }

    /**
     * 按照时间倒序排序，查询最近的访客信息
     *
     * @param userId  登录用户id
     * @param page   页数
     * @param pageSize  每页显示的条数
     * @return
     */
    @Override
    public PageInfo<Visitors> topVisitor(Long userId, Integer page, Integer pageSize) {
        PageRequest pageRequest = PageRequest.of(page, pageSize,Sort.by(Sort.Order.desc("date")));
        Query query = Query.query(Criteria.where("userId").is(userId)).with(pageRequest);
        List<Visitors> visitorsList = queryList(query,userId);
        PageInfo<Visitors> pageInfo = new PageInfo<>();
        pageInfo.setPageNum(page);
        pageInfo.setPageSize(pageSize);
        pageInfo.setRecords(visitorsList);

        // 记录当前的时间到redis中, 在首页查询时, 就可以在这个时间之后查询了
        String redisKey = VISITOR_REDIS_KEY;
        String hashKey = String.valueOf(userId);
        String value = String.valueOf(System.currentTimeMillis());
        redisTemplate.opsForHash().put(redisKey, hashKey, value);

        return pageInfo;

    }

    private List<Visitors> queryList(Query query, Long userId) {
        List<Visitors> visitorsList = mongoTemplate.find(query, Visitors.class);
        // 查询每个访客的得分
        for (Visitors visitors : visitorsList) {
            Query queryScore = Query.query(Criteria.where("toUserId").is(userId).and("userId").is(visitors.getVisitorUserId()));
            RecommendUser recommendUser = mongoTemplate.findOne(queryScore, RecommendUser.class);
            if (ObjectUtil.isNotEmpty(recommendUser)){
                visitors.setScore(recommendUser.getScore());
            }else {
                // 默认得分
                visitors.setScore(90D);
            }
        }
        return visitorsList;
    }
}
